package woa.linkedlist;

import java.util.HashMap;
import java.util.Map;

/**
 * 复制带随机指针的链表
 * 给你一个长度为 n 的链表，每个节点包含一个额外增加的随机指针 random ，该指针可以指向链表中的任何节点或空节点。
 * 构造这个链表的 深拷贝。 深拷贝应该正好由 n 个 全新 节点组成，其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点，并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点 。
 * 例如，如果原链表中有 X 和 Y 两个节点，其中 X.random --> Y 。那么在复制链表中对应的两个节点 x 和 y ，同样有 x.random --> y 。
 * 返回复制链表的头节点。
 * 用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示：
 * val：一个表示 Node.val 的整数。
 * random_index：随机指针指向的节点索引（范围从 0 到 n-1）；如果不指向任何节点，则为  null 。
 * 你的代码 只 接受原链表的头节点 head 作为传入参数。
 *
 * 输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
 * 输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]
 *
 * @author wangpeng
 * @date 2021/11/22
 */
public class CopyRandomListProblem {

    static class Node {
        int val;
        Node next;
        Node random;

        public Node(int val) {
            this.val = val;
            this.next = null;
            this.random = null;
        }
    }

    public static Node copyRandomList(Node head) {
        Node sentinel = new Node(0);
        sentinel.next = head;
        Node cur = sentinel;
        Node p = head;
        Map<Node, Node> randomMap = new HashMap<Node, Node>(16);
        while (p != null) {
            Node newNode;
            if (randomMap.containsKey(p)) {
                newNode = randomMap.get(p);
            } else {
                newNode = new Node(p.val);
                randomMap.put(p, newNode);
            }
            cur.next = newNode;
            if (p.random != null) {
                if (randomMap.containsKey(p.random)) {
                    newNode.random = randomMap.get(p.random);
                } else {
                    newNode.random = new Node(p.random.val);
                    randomMap.put(p.random, newNode.random);
                }
            }
            cur = cur.next;
            p = p.next;
        }

        return sentinel.next;
    }

    public static void main(String[] args) {

    }
}
